home *** CD-ROM | disk | FTP | other *** search
/ Sun Solutions 1997 April to September / Sun Solutions CD - APR '97 - SEP '97 (704-3778-12 Rev. H)(Sun Microsystems, Inc.)(1997).iso / products / bin / httpd / src / http_ipc.c < prev    next >
C/C++ Source or Header  |  1995-05-18  |  6KB  |  285 lines
  1. /*
  2.  * http_ipc.c                                                            
  3.  *    Handles the file descriptor passing.  Should work under BSD and    
  4.  *   SYSV.                                                               
  5.  *
  6.  * All code contained herein is covered by the Copyright as distributed
  7.  * in the README file in the main directory of the distribution of 
  8.  * NCSA HTTPD.
  9.  *
  10.  * Based (with modifications) on code by W. Richard Stevens, 
  11.  *    in _Advanced Programming in the UNIX Environment_
  12.  *                                                                       
  13.  * 03-06-95  blong
  14.  *    Added #ifdefs to handle not using this at all for machines like 
  15.  *     Linux.
  16.  *
  17.  * 04-17-95  blong
  18.  *    Added NEED_SPIPE with s_pipe from Stevens _Unix Network Programming_
  19.  *     for SVR3.2 systems without socketpair.
  20.  */
  21.  
  22. #ifndef NO_PASS
  23.  
  24. #include "httpd.h"
  25.  
  26. #include <sys/types.h>
  27. #if defined(FD_BSD)
  28. #include <sys/uio.h>
  29. #include <errno.h>
  30. #include <stddef.h>
  31. #elif defined(FD_SYSV)
  32. #ifdef NEED_SPIPE
  33. #define SPX_DEVICE    "/dev/spx"
  34. #include <sys/stream.h>
  35. #endif
  36. #include <stropts.h>
  37. #endif
  38. #include "new.h"
  39.  
  40. #ifdef FD_SYSV
  41.  
  42. int pass_fd(int clifd, int fd) {
  43.     char buf[2];
  44.  
  45.     buf[0] = 0;
  46.     if (fd < 0) {
  47.     buf[1] = -fd;
  48.     if (buf[1] == 0)
  49.         buf[1] = 1;
  50.     } else {
  51.     buf[1] = 0;
  52.     }
  53.  
  54.     if (write(clifd, buf, 2) != 2) {
  55.     fprintf(stderr,"pass_fd: write failed\n");
  56.     perror("pass_fd");
  57.     return -1;
  58.     }
  59.     if (fd >= 0)
  60.     if (ioctl(clifd, I_SENDFD, fd) < 0) {
  61.         fprintf(stderr,"pass_fd: ioctl failed\n");
  62.         perror("ioctl");
  63.         return -1;
  64.     }
  65.     return 0;
  66. } 
  67.  
  68. int recv_fd(int servfd) {
  69.     int  newfd, nread, flag, status;
  70.     char  *ptr, buf[IOBUFSIZE];
  71.     struct strbuf dat;
  72.     struct strrecvfd recvfd;
  73.  
  74.     status = -1;
  75.     for ( ; ; ) {
  76.     dat.buf = buf;
  77.     dat.maxlen = IOBUFSIZE;
  78.     flag = 0;
  79.     if (getmsg(servfd, NULL, &dat, &flag) < 0) {
  80.         fprintf(stderr,"httpd:getmsg error recv_fd\n");
  81.         perror("getmsg");
  82.         exit(1);
  83.     }
  84.     nread = dat.len;
  85.     if (nread == 0) {
  86.         fprintf(stderr,"httpd: connection closed by server\n");
  87.         exit(1);
  88.     }
  89.     
  90.     for (ptr = buf; ptr < &buf[nread]; ) {
  91.         if (*ptr++ == 0) {
  92.         if (ptr != &buf[nread-1]) {
  93.             fprintf(stderr,"httpd: message format error recv_fd\n");
  94.             perror("recv_fd");
  95.             exit(1);
  96.         }
  97.         status = *ptr & 255;
  98.         if (status == 0) {
  99.             if (ioctl(servfd, I_RECVFD, &recvfd) < 0)
  100.             return -1;
  101.             newfd = recvfd.fd;
  102.         } else 
  103.             newfd = -status;
  104.         nread -= 2;
  105.         }
  106.     }
  107.     if (status >= 0)
  108.         return (newfd);
  109.     }
  110. } 
  111.  
  112. #ifdef NEED_SPIPE
  113.  
  114. int s_pipe(int fd[2]) {
  115.   struct strfdinsert ins;
  116.   queue_t  *pointer;
  117.  
  118.   /*
  119.    * First open the stream clone device "/dev/spx" twice,
  120.    * obtaining the two file descriptors.
  121.    */
  122.  
  123.   if ((fd[0] = open(SPX_DEVICE, O_RDWR)) < 0) 
  124.     return -1;
  125.   if ((fd[1] = open(SPX_DEVICE, O_RDWR)) < 0)
  126.     return -1;
  127.  
  128.   /* 
  129.    * Now link these two streams together with an I_FDINSERT ioctl
  130.    */
  131.  
  132.   ins.ctlbuf.buf = (char *) &pointer;
  133.   ins.ctlbuf.maxlen = sizeof(queue_t *);
  134.   ins.ctlbuf.len = sizeof(queue_t *);
  135.  
  136.   ins.databuf.buf = (char *) 0;
  137.   ins.databuf.len = -1;
  138.   ins.databuf.maxlen = 0;
  139.  
  140.   ins.fildes = fd[1];
  141.   ins.flags = 0;
  142.   ins.offset = 0;
  143.  
  144.   if (ioctl(fd[0], I_FDINSERT, (char *) &ins) < 0) {
  145.     close(fd[0]);
  146.     close(fd[1]);
  147.     return -1;
  148.   }
  149.  
  150.   return 0;
  151.  
  152. }
  153.  
  154. #endif /* NEED_SPIPE */
  155.  
  156. #endif /* FD_SYSV */
  157.  
  158. #ifdef FD_BSD
  159.  
  160. #ifdef FD_BSDRENO
  161. static struct cmsghdr *cmptr = NULL;
  162. #define CONTROLLEN (sizeof(struct cmsghdr) + sizeof(int))
  163. #endif
  164.  
  165. int pass_fd(int clifd, int fd) {
  166.   struct iovec iov[1];
  167.   struct msghdr msg;
  168.   char  buf[2];
  169.  
  170.   iov[0].iov_base = buf;
  171.   iov[0].iov_len = 2;
  172.   msg.msg_iov = iov;
  173.   msg.msg_iovlen = 1;
  174.   msg.msg_name = NULL;
  175.   msg.msg_namelen = 0;
  176.   if (fd < 0) {
  177. #ifdef FD_BSDRENO
  178.     msg.msg_control = NULL;
  179.     msg.msg_controllen = 0;
  180. #else
  181.     msg.msg_accrights = NULL;
  182.     msg.msg_accrightslen = 0;
  183. #endif
  184.     buf[1] = -fd;
  185.     if (buf[1] == 0)
  186.       buf[1] = 1;
  187.   } else {
  188. #ifdef FD_BSDRENO
  189.     if (cmptr == NULL && (cmptr = malloc(CONTROLLEN)) == NULL)
  190.       return -1;
  191.     cmptr->cmsg_level = SOL_SOCKET;
  192.     cmptr->cmsg_type = SCM_RIGHTS;
  193.     cmptr->cmsg_len = CONTROLLEN;
  194.     msg.msg_control = (caddr_t) cmptr;
  195.     msg.msg_controllen = CONTROLLEN;
  196.     *(int *)CMSG_DATA(cmptr) = fd;
  197. #else
  198.     msg.msg_accrights = (caddr_t) &fd;
  199.     msg.msg_accrightslen = sizeof(int);
  200. #endif
  201.     buf[1] = 0;
  202.   }
  203.   buf[0] = 0;
  204.  
  205. /*  if (write(clifd, buf, 2) != 2)
  206.    return -1; */
  207.  
  208.   if (sendmsg(clifd, &msg, 0) != 2)
  209.     return -1;
  210.   
  211.   return 0;
  212. }
  213.  
  214.  
  215. int recv_fd(int servfd) {
  216.   int  nread, status;
  217.   int newfd = -1;
  218.   char *ptr, buf[IOBUFSIZE];
  219.   struct iovec  iov[1];
  220.   struct msghdr  msg;
  221.   
  222.   status = -1;
  223.   for ( ; ; ) {
  224.     iov[0].iov_base = buf;
  225.     iov[0].iov_len = sizeof(buf);
  226.     msg.msg_iov = iov;
  227.     msg.msg_iovlen = 1;
  228.     msg.msg_name = NULL;
  229.     msg.msg_namelen = 0;
  230. #ifdef FD_BSDRENO
  231.     if (cmptr == NULL && (cmptr = malloc(CONTROLLEN)) == NULL)
  232.       return -1;
  233.     msg.msg_control = (caddr_t) cmptr;
  234.     msg.msg_controllen = CONTROLLEN;
  235. #else
  236.     msg.msg_accrights = (caddr_t) &newfd;
  237.     msg.msg_accrightslen = sizeof(int);
  238. #endif
  239.     
  240.     if ((nread = recvmsg(servfd, &msg, 0)) < 0) {
  241.       fprintf(stderr,"httpd: recvmsg error");
  242.       perror("recvmsg");
  243.       exit(1);
  244.     } else if (nread == 0) {
  245.       fprintf(stderr, "httpd: connection closed by server");
  246.       exit(1);
  247.     }
  248.     
  249.     for (ptr = buf; ptr < &buf[nread]; ) {
  250.       if (*ptr++ == 0) {
  251.     if (ptr != &buf[nread-1]) {
  252.       fprintf(stderr, "httpd:message format error");
  253.       exit(1);
  254.     }
  255.     status = *ptr & 255;
  256.     if (status == 0) {
  257. #ifdef FD_BSDRENO
  258.       if (msg.msg_controllen != CONTROLLEN) 
  259. #else
  260.         if (msg.msg_accrightslen != sizeof(int)) 
  261. #endif    
  262.           {
  263.         fprintf(stderr, "httpd: status = 0 but no fd");
  264.         exit(1);
  265.           }
  266. #ifdef FD_BSDRENO
  267.       newfd = *(int *)CMSG_DATA(cmptr);
  268. #endif
  269.     } else 
  270.       newfd = -status;
  271.       nread -= 2;
  272.       }
  273.     }
  274.     if (nread > 0)
  275.       return -1;
  276.     if (status >= 0)
  277.       return newfd;
  278.   }
  279. }
  280.   
  281.  
  282. #endif
  283.  
  284. #endif 
  285.